package com.shujia.scala

import scala.io.Source

object Demo24WordCount {
  def main(args: Array[String]): Unit = {

    //1、读取文件
    val lines: List[String] = Source
      .fromFile("data/words.txt") //读取文件
      .getLines() //获取所有行
      .toList //转换成list集合

    //2、将一行中的多个单词拆分成多行
    val words: List[String] = lines.flatMap((line: String) => line.split(","))

    //3、按照单词分组
    val groupBy: Map[String, List[String]] = words.groupBy(word => word)


    //4、计算每一个单词的数量
    val wordCount: Map[String, Int] = groupBy.map((kv: (String, List[String])) => {
      //一个单词
      val word: String = kv._1
      //同一个单词构建的集合
      val values: List[String] = kv._2

      //获取单词的数量
      val count: Int = values.length

      //返回单词和数量
      (word, count)
    })


    wordCount.foreach(println)


    /**
      * 简写
      *
      */

    //链式调用
    Source
      .fromFile("data/words.txt") //读取文件
      .getLines() //获取所有行
      .toList //转换成list
      .flatMap(line => line.split(",")) //将数据转换成多行
      .groupBy(word => word) //按照单词分组
      .map(kv => (kv._1, kv._2.length)) //统计单词的数量
      .foreach(println) //打印结果


  }

}
